import argparse


parser=argparse.ArgumentParser(description="Pytorch Experiment for distill")

#train the teacher model
parser.add_argument('--train_batch_size',type=int,default=4)
parser.add_argument('--test_batch_size',type=int,default=4)
parser.add_argument('--epoches',type=int,default=500)
parser.add_argument('--lr',type=float,default=0.001)
parser.add_argument('--momentum',type=float,default=0.9)
parser.add_argument('--weight_decay',type=float,default=5e-3)
parser.add_argument('--seed',type=int,default=123)
parser.add_argument('--use_cuda',type=bool,default=True)
parser.add_argument('--train_data_path',type=str,default='F:/datasets_dog/train')
parser.add_argument('--test_data_path',type=str,default='F:/datasets_dog/val')
parser.add_argument('--resume',type=bool,default=True)
parser.add_argument('--save_model',type=str,default='./weights/teacher')
parser.add_argument('--pretrained_model',type=str,default="C:/Users/lth/Desktop/distill/weights/student/best.pth")
parser.add_argument('--project_name',type=str,default="模型蒸馏测试")
parser.add_argument('--save',type=bool,default=True)
parser.add_argument('--use_aug',type=bool,default=True)
